package com.ruoyi.framework.config;

import java.io.IOException;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.HashSet;
import java.util.List;

import com.baomidou.mybatisplus.autoconfigure.MybatisPlusProperties;
import com.baomidou.mybatisplus.core.MybatisConfiguration;
import com.baomidou.mybatisplus.extension.spring.MybatisSqlSessionFactoryBean;
import com.ruoyi.framework.datasource.DynamicDataSource;
import org.apache.ibatis.session.SqlSessionFactory;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Qualifier;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.context.annotation.Primary;
import org.springframework.core.env.Environment;
import org.springframework.core.io.Resource;
import org.springframework.core.io.support.PathMatchingResourcePatternResolver;
import org.springframework.core.io.support.ResourcePatternResolver;
import org.springframework.core.type.classreading.CachingMetadataReaderFactory;
import org.springframework.core.type.classreading.MetadataReader;
import org.springframework.core.type.classreading.MetadataReaderFactory;
import org.springframework.util.ClassUtils;

/**
* Mybatis支持*匹配扫描包
*
* @author ruoyi
*/
@Configuration
public class MyBatisConfig
{
   @Autowired
   private Environment env;
   @Autowired
   private MybatisPlusProperties properties;

   static final String DEFAULT_RESOURCE_PATTERN = "**/*.class";

   public static String setTypeAliasesPackage(String typeAliasesPackage)
   {
       ResourcePatternResolver resolver = (ResourcePatternResolver) new PathMatchingResourcePatternResolver();
       MetadataReaderFactory metadataReaderFactory = new CachingMetadataReaderFactory(resolver);
       List<String> allResult = new ArrayList<String>();
       try
       {
           for (String aliasesPackage : typeAliasesPackage.split(","))
           {
               List<String> result = new ArrayList<String>();
               aliasesPackage = ResourcePatternResolver.CLASSPATH_ALL_URL_PREFIX
                       + ClassUtils.convertClassNameToResourcePath(aliasesPackage.trim()) + "/" + DEFAULT_RESOURCE_PATTERN;
               Resource[] resources = resolver.getResources(aliasesPackage);
               if (resources != null && resources.length > 0)
               {
                   MetadataReader metadataReader = null;
                   for (Resource resource : resources)
                   {
                       if (resource.isReadable())
                       {
                           metadataReader = metadataReaderFactory.getMetadataReader(resource);
                           try
                           {
                               result.add(Class.forName(metadataReader.getClassMetadata().getClassName()).getPackage().getName());
                           }
                           catch (ClassNotFoundException e)
                           {
                               e.printStackTrace();
                           }
                       }
                   }
               }
               if (result.size() > 0)
               {
                   HashSet<String> hashResult = new HashSet<String>(result);
                   allResult.addAll(hashResult);
               }
           }
           if (allResult.size() > 0)
           {
               typeAliasesPackage = String.join(",", (String[]) allResult.toArray(new String[0]));
           }
           else
           {
               throw new RuntimeException("mybatis typeAliasesPackage 路径扫描错误,参数typeAliasesPackage:" + typeAliasesPackage + "未找到任何包");
           }
       }
       catch (IOException e)
       {
           e.printStackTrace();
       }
       return typeAliasesPackage;
   }

   public Resource[] resolveMapperLocations(String[] mapperLocations)
   {
       ResourcePatternResolver resourceResolver = new PathMatchingResourcePatternResolver();
       List<Resource> resources = new ArrayList<Resource>();
       if (mapperLocations != null)
       {
           for (String mapperLocation : mapperLocations)
           {
               try
               {
                   Resource[] mappers = resourceResolver.getResources(mapperLocation);
                   resources.addAll(Arrays.asList(mappers));
               }
               catch (IOException e)
               {
                   // ignore
               }
           }
       }
       return resources.toArray(new Resource[resources.size()]);
   }

   // @Bean
   // public SqlSessionFactory sqlSessionFactory(DataSource dataSource) throws Exception
   // {
   //     String typeAliasesPackage = env.getProperty("mybatis-plus.typeAliasesPackage");
   //     String mapperLocations = env.getProperty("mybatis-plus.mapperLocations");
   //     // String configLocation = env.getProperty("mybatis-plus.configLocation");
   //     typeAliasesPackage = setTypeAliasesPackage(typeAliasesPackage);
   //     VFS.addImplClass(SpringBootVFS.class);
   //
   //     final SqlSessionFactoryBean sessionFactory = new SqlSessionFactoryBean();
   //     sessionFactory.setDataSource(dataSource);
   //     sessionFactory.setTypeAliasesPackage(typeAliasesPackage);
   //     sessionFactory.setMapperLocations(resolveMapperLocations(StringUtils.split(mapperLocations, ",")));
   //     sessionFactory.setConfigLocation(new DefaultResourceLoader().getResource("classpath:mybatis/mybatis-config.xml"));
   //     return sessionFactory.getObject();
   // }

    /**
     * 创建动态数据源的SqlSessionFactory，传入的是动态数据源.
     * Primary 这个注解很重要，如果项目中存在多个SqlSessionFactory，这个注解一定要加上.
     */
    @Primary
    @Bean("sqlSessionFactory")
    public SqlSessionFactory sqlSessionFactoryBean(@Qualifier("dynamicDataSource") DynamicDataSource
                                                           dynamicDataSource) throws Exception {

        MybatisSqlSessionFactoryBean sqlSessionFactoryBean = new MybatisSqlSessionFactoryBean();
        sqlSessionFactoryBean.setDataSource(dynamicDataSource);

        MybatisConfiguration configuration = properties.getConfiguration();
        sqlSessionFactoryBean.setMapperLocations(properties.resolveMapperLocations());
        sqlSessionFactoryBean.setConfiguration(configuration);
        sqlSessionFactoryBean.setTypeAliasesPackage(properties.getTypeAliasesPackage());
        sqlSessionFactoryBean.setGlobalConfig(properties.getGlobalConfig());
        return sqlSessionFactoryBean.getObject();
    }
}
